package cn.tedu.boot02.controller;

import cn.tedu.boot02.entity.Product;
import cn.tedu.boot02.utils.DBUtil;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;

@RestController
public class ProductController {
    @RequestMapping("/insert")
    public String insert(Product product){
        System.out.println("product = " + product);
        try (Connection c= DBUtil.getConnection()){
            String sql="INSERT INTO product VALUES(NULL,?,?,?)";
            PreparedStatement ps=c.prepareStatement(sql);
            ps.setString(1,product.getTitle());
            ps.setDouble(2,product.getPrice());
            ps.setInt(3,product.getNum());
            ps.executeUpdate();

        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return "添加完成";
    }

    @RequestMapping("/select")
    public String select(){
        //创建集合来装查询到的所有商品对象
        ArrayList<Product> list=new ArrayList<>();
        try (Connection c=DBUtil.getConnection()){
            String sql="SELECT * FROM product";
            PreparedStatement ps=c.prepareStatement(sql);
            ResultSet rs=ps.executeQuery();
            while (rs.next()){
                Product product=new Product();
                product.setId(rs.getInt("id"));
                product.setTitle(rs.getString("title"));
                product.setPrice(rs.getDouble("price"));
                product.setNum(rs.getInt("num"));
                list.add(product);
            }
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        String html="<table border=1>";
        html+="<tr><th>id</th><th>标题</th><th>价格</th><th>库存</th><th>操作</th></tr>";
        for(Product p:list){
            html+="<tr>";
            html+="<td>"+p.getId()+"</td>";
            html+="<td>"+p.getTitle()+"</td>";
            html+="<td>"+p.getPrice()+"</td>";
            html+="<td>"+p.getNum()+"</td>";
            html+="<td><a href='/delete?id="+p.getId()+"'>删除</a></td>";
            html+="</tr>";
        }
        html+="<tr style='text-align: center'><td colspan=5><a href='/'>返回首页</a></td></tr>";
        html+="</table>";
        return html;
    }

    @RequestMapping("/delete")
    public String delete(int id){
        System.out.println("id = " + id);
        try (Connection con=DBUtil.getConnection()){
            String sql= "DELETE FROM product WHERE id=?";
            PreparedStatement ps=con.prepareStatement(sql);
            ps.setInt(1,id);
            ps.executeUpdate();
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }

        return "删除成功<a href='/select'>返回列表页面</a>";
    }

    @RequestMapping("/update")
    public String update(Product p){
        try (Connection con=DBUtil.getConnection()){
            String sql="UPDATE product SET title=?,price=?,num=? WHERE id=?";
            PreparedStatement ps=con.prepareStatement(sql);
            ps.setString(1,p.getTitle());
            ps.setDouble(2,p.getPrice());
            ps.setInt(3,p.getNum());
            ps.setInt(4,p.getId());
            ps.executeUpdate();
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return "修改成功<a href='/select'>返回列表页面</a>";
    }
}
